package com.heima.leetcode.practice;

import java.util.HashMap;

/**
 * leetcode 105. 从前序与中序遍历序列构造二叉树
 *
 * @author 勾新杰
 * @version 1.0
 * @date 2025/1/10 23:55
 */
public class E105 {

    /**
     * 存放中序遍历的节点值和索引
     */
    private final HashMap<Integer, Integer> map = new HashMap<>();

    /**
     * <h3>方法一：递归，分而治之</h3>
     *
     * @param preorder 先序遍历
     * @param inorder  中序遍历
     * @return 构建出来的树
     */
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for (int i = 0; i < inorder.length; i++) map.put(inorder[i], i);
        return buildTree(
                preorder,
                0,
                preorder.length - 1,
                inorder,
                0,
                inorder.length - 1);
    }

    /**
     * 递归构建树
     *
     * @param preorder 当前树先序遍历
     * @param preBegin 当前树的先序遍历的起始索引
     * @param preEnd   当前树的先序遍历的结束索引
     * @param inorder  当前树的中序遍历
     * @param inBegin  当前树的中序遍历的起始索引
     * @param inEnd    当前树的中序遍历的结束索引
     * @return 构建出来的树
     */
    private TreeNode buildTree(
            int[] preorder,
            int preBegin,
            int preEnd,
            int[] inorder,
            int inBegin,
            int inEnd) {
        // 1. 递归结束条件：当前树遍历结果为空（当前树前序遍历和中序遍历对应的）
        if (preBegin > preEnd) return null;
        // 2. 找到当前树的根节点的值
        int nodeValue = preorder[preBegin];
        TreeNode node = new TreeNode(nodeValue);
        Integer median = map.get(nodeValue);
        int leftLength = median - inBegin;
        // 3. 构建左子树、右子树
        node.left = buildTree(
                preorder,
                preBegin + 1,
                preBegin + leftLength,
                inorder,
                inBegin,
                median - 1);
        node.right = buildTree(
                preorder,
                preBegin + leftLength + 1,
                preEnd,
                inorder,
                median + 1,
                inEnd);
        // 4. 返回当前树
        return node;
    }
}
